home *** CD-ROM | disk | FTP | other *** search
/ Kit PC World De Ampliacion De Windows 95 / Kit PC World de ampliacion de Windows 95.iso / internet / sweeper / samples / progress / progress.cpp < prev    next >
C/C++ Source or Header  |  1995-12-05  |  12KB  |  339 lines
  1. // ===========================================================================
  2. // File: PROGRESS.CPP
  3. // 
  4. // Copyright 1995 - 1996 Microsoft Corporation.  All Rights Reserved.
  5. // ===========================================================================
  6. #include <urlmon.h>
  7. #include "resource.h"
  8.  
  9. // %%Classes: ----------------------------------------------------------------
  10. class CDownload {
  11.   public:
  12.     CDownload(LPCWSTR szURL);
  13.     HRESULT DoDownload(HWND,HWND,HWND);
  14.     LPCWSTR              m_url;
  15.  
  16.   private:
  17.     IMoniker*            m_pmk;
  18.     IBindCtx*            m_pbc;
  19.     IBindStatusCallback* m_pbsc;
  20. };
  21.  
  22. class CBindStatusCallback : public IBindStatusCallback {
  23.   public:
  24.     // IUnknown methods
  25.     STDMETHODIMP    QueryInterface(REFIID riid,void ** ppv);
  26.     STDMETHODIMP_(ULONG)    AddRef()    { return m_cRef++; }
  27.     STDMETHODIMP_(ULONG)    Release()   { if (--m_cRef == 0) { delete this; return 0; } return m_cRef; }
  28.  
  29.     // IBindStatusCallback methods
  30.     STDMETHODIMP    GetBindInfo(DWORD* pgrfBINDF, BINDINFO* pbindinfo);
  31.     STDMETHODIMP    OnStartBinding(IBinding* pbinding);
  32.     STDMETHODIMP    GetPriority(LONG* pnPriority);
  33.     STDMETHODIMP    OnProgress(ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode,
  34.                         LPCWSTR pwzStatusText);
  35.     STDMETHODIMP    OnDataAvailable(DWORD grfBSCF, DWORD dwSize, FORMATETC *pFmtetc,
  36.                         REFIID riid, IUnknown* punk);
  37.     STDMETHODIMP    OnLowResource(DWORD dwReserved);
  38.     STDMETHODIMP    OnStopBinding(HRESULT hrResult);
  39.  
  40.     // constructors/destructors
  41.     CBindStatusCallback(HWND hwndStatus, HWND hwndProgress, HWND hwndText);
  42.  
  43.     inline void SetStatus(LPCWSTR szStatus)     { SetWndText(m_hwndStatus, szStatus); }
  44.     void SetProgress(LPCWSTR szProgress)        { SetWndText(m_hwndProgress, szProgress); }
  45.     void SetWndText(HWND hwnd, LPCWSTR szText);
  46.  
  47.     // data members
  48.     DWORD           m_cRef;
  49.     IBinding*       m_pbinding;
  50.     IStream*        m_pstm;
  51.     HWND            m_hwndStatus;
  52.     HWND            m_hwndProgress;
  53.     HWND            m_hwndText;
  54.     DWORD           m_cbOld;
  55. };
  56.  
  57.  
  58. // ---------------------------------------------------------------------------
  59. // %%Function: CBindStatusCallback::CBindStatusCallback
  60. // ---------------------------------------------------------------------------
  61. CBindStatusCallback::CBindStatusCallback(HWND hwndStatus, HWND hwndProgress, HWND hwndText)
  62. {
  63.     m_hwndStatus = hwndStatus;
  64.     m_hwndProgress = hwndProgress;
  65.     m_hwndText = hwndText;
  66.     m_pbinding = NULL;
  67.     m_pstm = NULL;
  68.     m_cRef = 1;
  69.     m_cbOld = 0;
  70. }  // CBindStatusCallback
  71.  
  72. // ---------------------------------------------------------------------------
  73. // %%Function: CBindStatusCallback::~CBindStatusCallback
  74. // ---------------------------------------------------------------------------
  75. CBindStatusCallback::~CBindStatusCallback()
  76. {
  77.     if (m_pstm)
  78.         m_pstm->Release();
  79.     if (m_pbinding)
  80.         m_pbinding->Release();
  81. }  // ~CBindStatusCallback
  82.  
  83. // ---------------------------------------------------------------------------
  84. // %%Function: CBindStatusCallback::SetWndText
  85. // ---------------------------------------------------------------------------
  86.  inline void
  87. CBindStatusCallback::SetWndText(HWND hwnd,LPCWSTR szText)
  88. {
  89.     if (IsWindow(hwnd))
  90.         {
  91.         char    rgchBuf[MAX_PATH];
  92.         WideCharToMultiByte(CP_ACP, 0, szText, -1, rgchBuf, MAX_PATH, 0, 0);
  93.     SetWindowText(hwnd, rgchBuf);
  94.     }
  95. }  // CBindStatusCallback::SetWndText
  96.  
  97. // ---------------------------------------------------------------------------
  98. // %%Function: CBindStatusCallback::QueryInterface
  99. // ---------------------------------------------------------------------------
  100.  STDMETHODIMP
  101. CBindStatusCallback::QueryInterface(REFIID riid, void** ppv)
  102. { 
  103.     *ppv = NULL;
  104.     
  105.     if (riid==IID_IUnknown || riid==IID_IBindStatusCallback)
  106.         {
  107.         *ppv = this;
  108.         AddRef();
  109.         return S_OK;
  110.         }
  111.     return E_NOINTERFACE;
  112. }  // CBindStatusCallback::QueryInterface
  113.  
  114. // ---------------------------------------------------------------------------
  115. // %%Function: CBindStatusCallback::GetBindInfo
  116. // ---------------------------------------------------------------------------
  117.  STDMETHODIMP
  118. CBindStatusCallback::GetBindInfo(DWORD* pgrfBINDF, BINDINFO* pbindInfo)
  119. {
  120.     *pgrfBINDF = BINDF_ASYNCHRONOUS;
  121.     pbindInfo->cbSize = sizeof(BINDINFO);
  122.     pbindInfo->szExtraInfo = NULL;
  123.     memset(&pbindInfo->stgmedPost, 0, sizeof(STGMEDIUM));
  124.     return S_OK;
  125. }  // CBindStatusCallback::GetBindInfo
  126.  
  127. // ---------------------------------------------------------------------------
  128. // %%Function: CBindStatusCallback::OnStartBinding
  129. // ---------------------------------------------------------------------------
  130.  STDMETHODIMP
  131. CBindStatusCallback::OnStartBinding(IBinding* pbinding)
  132. {
  133.     if (m_pbinding != NULL)
  134.         m_pbinding->Release();
  135.     m_pbinding = pbinding;
  136.     if (m_pbinding != NULL)
  137.         {
  138.         m_pbinding->AddRef();
  139.         SetStatus(L"Status: Starting to bind...");
  140.         }
  141.     return S_OK;
  142. }  // CBindStatusCallback::OnStartBinding
  143.  
  144. // ---------------------------------------------------------------------------
  145. // %%Function: CBindStatusCallback::GetPriority
  146. // ---------------------------------------------------------------------------
  147.  STDMETHODIMP
  148. CBindStatusCallback::GetPriority(LONG* pnPriority) 
  149. {
  150.     return E_NOTIMPL;
  151. }  // CBindStatusCallback::GetPriority
  152.  
  153. // ---------------------------------------------------------------------------
  154. // %%Function: CBindStatusCallback::OnProgress
  155. // ---------------------------------------------------------------------------
  156.  STDMETHODIMP
  157. CBindStatusCallback::OnProgress(ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
  158. {
  159.     char    sz[255];
  160.     WideCharToMultiByte(CP_ACP, 0, szStatusText,-1, sz, 255, 0,0);
  161.  
  162.     char    msg[100];
  163.     WCHAR   out[200];
  164.  
  165.     wsprintf(msg,"Progress: %s %d of %d ", sz, ulProgress, ulProgressMax);
  166.     MultiByteToWideChar(CP_ACP, 0, msg, -1, out, sizeof(out));
  167.  
  168.     SetProgress(out);
  169.     return(NOERROR);
  170. }  // CBindStatusCallback
  171.  
  172. // ---------------------------------------------------------------------------
  173. // %%Function: CBindStatusCallback::OnDataAvailable
  174. // ---------------------------------------------------------------------------
  175.  STDMETHODIMP
  176. CBindStatusCallback::OnDataAvailable(DWORD grfBSCF, DWORD dwSize, FORMATETC *pFmtetc, REFIID riid, IUnknown *punk) 
  177. {
  178.     // Assert(riid == IID_IStream);
  179.     if (!m_pstm)
  180.         m_pstm = (IStream *)punk;
  181.  
  182.     if (dwSize > m_cbOld)
  183.     {
  184.         DWORD dwRead = dwSize - m_cbOld;
  185.         DWORD dwActuallyRead;
  186.         char * lp = new char[dwRead];
  187.         m_pstm->Read(lp,dwRead,&dwActuallyRead);
  188.         SetWindowText(m_hwndText,lp);
  189.         delete lp;
  190.         m_cbOld = dwSize;
  191.     }
  192.     return S_OK;
  193. }  // CBindStatusCallback::OnDataAvailable
  194.  
  195. // ---------------------------------------------------------------------------
  196. // %%Function: CBindStatusCallback::OnLowResource
  197. // ---------------------------------------------------------------------------
  198.  STDMETHODIMP
  199. CBindStatusCallback::OnLowResource(DWORD dwReserved)
  200. {
  201.     return E_NOTIMPL;
  202. }  // CBindStatusCallback::OnLoadResource
  203.  
  204. // ---------------------------------------------------------------------------
  205. // %%Function: CBindStatusCallback::OnStopBinding
  206. // ---------------------------------------------------------------------------
  207.  STDMETHODIMP
  208. CBindStatusCallback::OnStopBinding(HRESULT hrStatus) 
  209. {
  210.     SetStatus(L"Status: Done.");
  211.     return S_OK;
  212. }  // CBindStatusCallback::OnStopBinding
  213.  
  214. // ---------------------------------------------------------------------------
  215. // %%Function: CDownload::CDownload
  216. // ---------------------------------------------------------------------------
  217. CDownload::CDownload(LPCWSTR szURL)
  218. {
  219.     m_url = szURL;
  220.     m_pmk = 0;
  221.     m_pbc = 0;
  222.     m_pbsc = 0;
  223. }  // CDownload
  224.  
  225. // ---------------------------------------------------------------------------
  226. // %%Function: CDownload::~CDownload
  227. // ---------------------------------------------------------------------------
  228. CDownload::~CDownload()
  229. {
  230.     if (m_pmk)
  231.         m_pmk->Release();
  232.     if (m_pbc)
  233.         m_pbc->Release();
  234.     if (m_pbsc)
  235.         m_pbsc->Release();
  236. }  // ~CDownload
  237.  
  238. // ---------------------------------------------------------------------------
  239. // %%Function: CDownload::DoDownload
  240. // ---------------------------------------------------------------------------
  241.  HRESULT
  242. CDownload::DoDownload(HWND hwndStatus, HWND hwndProgress, HWND hwndText)
  243. {
  244.     HRESULT hr =  CreateURLMoniker(NULL, m_url, &m_pmk);
  245.  
  246.     if( SUCCEEDED(hr) )
  247.         {
  248.         m_pbsc = new CBindStatusCallback(hwndStatus, hwndProgress, hwndText);
  249.         if (m_pbsc != NULL)
  250.             m_pbsc->AddRef();
  251.         else
  252.             hr = E_OUTOFMEMORY;
  253.     }
  254.  
  255.     if (SUCCEEDED(hr))
  256.         hr = CreateBindCtx(0, &m_pbc);
  257.  
  258.     if (SUCCEEDED(hr))
  259.         hr = m_pbc->RegisterObjectParam(OLESTR("BindStatusCallback"), m_pbsc);
  260.         
  261.     IStream* pstm;
  262.     if (SUCCEEDED(hr))
  263.         hr = m_pmk->BindToStorage(m_pbc, 0, IID_IStream, (void**)&pstm);
  264.  
  265.     return hr;
  266. }  // CDownload::DoDownload
  267.  
  268. // ---------------------------------------------------------------------------
  269. // %%Function: DialogProc
  270. // ---------------------------------------------------------------------------
  271.  BOOL CALLBACK
  272. DialogProc(HWND hwndDlg, UINT message, WPARAM wParam, LPARAM lParam)
  273. {
  274.     static CDownload* pcdl = NULL;
  275.  
  276.     switch(message)
  277.         {
  278.         case WM_INITDIALOG:
  279.             EnableWindow(GetDlgItem(hwndDlg, IDOK), TRUE);
  280.             EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), TRUE);
  281.             pcdl = reinterpret_cast<CDownload*>(lParam);
  282.             break;
  283.  
  284.         case WM_COMMAND:
  285.             {
  286.             switch (LOWORD(wParam))
  287.                 {
  288.                 case IDOK:
  289.                     {
  290.                     char rgchBuf[MAX_PATH];
  291.                     HWND hwndStatus = GetDlgItem(hwndDlg, IDC_DISPLAY);
  292.                     HWND hwndProgress = GetDlgItem(hwndDlg, IDC_PROGRESS);
  293.                     HWND hwndText = GetDlgItem(hwndDlg, IDC_TEXT);
  294.  
  295.                     EnableWindow(GetDlgItem(hwndDlg, IDOK), FALSE);
  296.                     EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), FALSE);
  297.  
  298.                     WideCharToMultiByte(CP_ACP, 0, pcdl->m_url, -1, rgchBuf, MAX_PATH, 0, 0);
  299.                 SetWindowText(hwndDlg, rgchBuf);
  300.  
  301.                     SetWindowText(hwndStatus, TEXT("Status: Inititating Bind..."));
  302.                     pcdl->DoDownload(hwndStatus, hwndProgress, hwndText);
  303.                     EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), TRUE);
  304.                     break;
  305.                     }
  306.  
  307.                 case IDCANCEL:
  308.                     EndDialog(hwndDlg,0);
  309.                     return 1;
  310.                 }
  311.             break;
  312.             }
  313.         }
  314.  
  315.     return 0;
  316. }  // DialogProc
  317.  
  318. // ---------------------------------------------------------------------------
  319. // %%Function: WinMain
  320. // ---------------------------------------------------------------------------
  321.  int WINAPI
  322. WinMain(HINSTANCE hinst, HINSTANCE hinstPrev, LPSTR szCmdLine, int nCmdShow)
  323. {
  324.     static WCHAR    szDefaultURL[] = L"http://www.home.msn.com";
  325.     WCHAR           rgwchCmdLine[MAX_PATH];
  326.  
  327.     MultiByteToWideChar(CP_ACP, 0, szCmdLine, -1, rgwchCmdLine, MAX_PATH);
  328.     
  329.     CDownload download(rgwchCmdLine[0] ? rgwchCmdLine : szDefaultURL);
  330.  
  331.     DialogBoxParam(hinst, MAKEINTRESOURCE(IDD_DIALOG1), HWND_DESKTOP,
  332.         (FARPROC)&DialogProc, reinterpret_cast<LPARAM>(&download));
  333.  
  334.     return 0;
  335. }  // WinMain
  336.  
  337. // EOF =======================================================================
  338.  
  339.